home *** CD-ROM | disk | FTP | other *** search
/ Mac Easy 2010 May / Mac Life Ubuntu.iso / casper / filesystem.squashfs / usr / lib / rhythmbox / plugins / artdisplay / LocalCoverArtSearch.py < prev    next >
Encoding:
Python Source  |  2009-04-07  |  6.6 KB  |  185 lines
  1. # -*- Mode: python; coding: utf-8; tab-width: 8; indent-tabs-mode: t; -*- 
  2. #
  3. # Copyright (C) 2006 - Ed Catmur <ed@catmur.co.uk>
  4. #
  5. # This program is free software; you can redistribute it and/or modify
  6. # it under the terms of the GNU General Public License as published by
  7. # the Free Software Foundation; either version 2, or (at your option)
  8. # any later version.
  9. #
  10. # The Rhythmbox authors hereby grant permission for non-GPL compatible
  11. # GStreamer plugins to be used and distributed together with GStreamer
  12. # and Rhythmbox. This permission is above and beyond the permissions granted
  13. # by the GPL license by which Rhythmbox is covered. If you modify this code
  14. # you may extend this exception to your version of the code, but you are not
  15. # obligated to do so. If you do not wish to do so, delete this exception
  16. # statement from your version.
  17. # 
  18. # This program is distributed in the hope that it will be useful,
  19. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  20. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21. # GNU General Public License for more details.
  22. #
  23. # You should have received a copy of the GNU General Public License
  24. # along with this program; if not, write to the Free Software
  25. # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301  USA.
  26.  
  27. import os
  28. import rhythmdb
  29. import gnomevfs
  30. import rb
  31. import gobject
  32.  
  33. IMAGE_NAMES = ["cover", "album", "albumart", ".folder", "folder"]
  34. LOAD_DIRECTORY_FLAGS = gnomevfs.FILE_INFO_GET_MIME_TYPE | gnomevfs.FILE_INFO_FORCE_FAST_MIME_TYPE | gnomevfs.FILE_INFO_FOLLOW_LINKS
  35. ITEMS_PER_NOTIFICATION = 10
  36. ART_SAVE_NAME = 'Cover.jpg'
  37. ART_SAVE_FORMAT = 'jpeg'
  38. ART_SAVE_SETTINGS = {"quality": "100"}
  39.  
  40. def file_root (f_name):
  41.     return os.path.splitext (f_name)[0].lower ()
  42.  
  43. def shared_prefix_length (a, b):
  44.     l = 0
  45.     while a[l] == b[l]:
  46.         l = l+1
  47.     return l
  48.  
  49.  
  50. class LocalCoverArtSearch:
  51.     def __init__ (self):
  52.         pass
  53.  
  54.     def _load_dir_cb (self, handle, files, exception, (results, on_search_completed_cb, entry, args)):
  55.         for f in files:
  56.             if f.mime_type is None:
  57.                 pass
  58.             elif f.mime_type.startswith ("image/") and f.permissions & gnomevfs.PERM_USER_READ:
  59.                 results.append (f.name)
  60.         if exception:
  61.             on_search_completed_cb (self, entry, results, *args)
  62.             if not issubclass (exception, gnomevfs.EOFError):
  63.                 print "Error reading \"%s\": %s" % (self.uri.parent, exception)
  64.  
  65.     def search (self, db, entry, on_search_completed_cb, *args):
  66.         self.uri = None
  67.         try:
  68.             self.uri = gnomevfs.URI (entry.get_playback_uri())
  69.         except TypeError:
  70.             pass
  71.  
  72.         if self.uri is None or self.uri.scheme == 'http':
  73.             print 'not searching for local art for %s' % (self.uri)
  74.             on_search_completed_cb (self, entry, [], *args)
  75.             return
  76.  
  77.         self.artist = db.entry_get (entry, rhythmdb.PROP_ARTIST)
  78.         self.album = db.entry_get (entry, rhythmdb.PROP_ALBUM)
  79.  
  80.         print 'searching for local art for %s' % (self.uri)
  81.         gnomevfs.async.load_directory (self.uri.parent, self._load_dir_cb, LOAD_DIRECTORY_FLAGS, ITEMS_PER_NOTIFICATION, data=([], on_search_completed_cb, entry, args))
  82.  
  83.     def search_next (self):
  84.         return False
  85.  
  86.     def get_best_match_urls (self, results):
  87.         # Compare lower case, without file extension
  88.         for name in [file_root (self.uri.short_name)] + IMAGE_NAMES:
  89.             for f_name in results:
  90.                 if file_root (f_name) == name:
  91.                     yield self.uri.parent.append_file_name (f_name)
  92.  
  93.         # look for file names containing the artist and album (case-insensitive)
  94.         # (mostly for jamendo downloads)
  95.         artist = self.artist.lower()
  96.         album = self.album.lower()
  97.         for f_name in results:
  98.             f_root = file_root (f_name).lower()
  99.             if f_root.find (artist) != -1 and f_root.find (album) != -1:
  100.                 yield self.uri.parent.append_file_name (f_name).path
  101.  
  102.         # if that didn't work, look for the longest shared prefix
  103.         # only accept matches longer than 2 to avoid weird false positives
  104.         match = (2, None)
  105.         for f_name in results:
  106.             pl = shared_prefix_length(f_name, self.uri.short_name)
  107.             if pl > match[0]:
  108.                 match = (pl, f_name)
  109.  
  110.         if match[1] is not None:
  111.             yield self.uri.parent.append_file_name (match[1]).path
  112.  
  113.     def pixbuf_save (self, plexer, pixbuf, uri):
  114.         gnomevfs.async.create (uri, plexer.send (), gnomevfs.OPEN_WRITE | gnomevfs.OPEN_TRUNCATE, False, 0644, gnomevfs.PRIORITY_DEFAULT)
  115.         yield None
  116.         _, (handle, result) = plexer.receive ()
  117.         if result:
  118.             print "Error creating \"%s\": %s" % (uri, result)
  119.             return
  120.         def pixbuf_cb (buf):
  121.             data = [buf]
  122.             status = []
  123.             def write_coro (w_plexer):
  124.                 while data:
  125.                     buf = data.pop ()
  126.                     handle.write (buf, w_plexer.send ())
  127.                     yield None
  128.                     _, (_, requested, result, written) = w_plexer.receive ()
  129.                     if result:
  130.                         print "Error writing \"%s\": %s" % (uri, result)
  131.                         status.insert (0, False)
  132.                         return
  133.                     if written < requested:
  134.                         data.insert (0, buf[written:])
  135.                 status.insert (0, True)
  136.             rb.Coroutine (write_coro).begin ()
  137.             while not status:
  138.                 gobject.main_context_default ().iteration ()
  139.             return status[0]
  140.         pixbuf.save_to_callback (pixbuf_cb, ART_SAVE_FORMAT, ART_SAVE_SETTINGS)
  141.         handle.close (plexer.send())
  142.         yield None
  143.         _, (_, result) = plexer.receive ()
  144.         if result:
  145.             print "Error closing \"%s\": %s" % (uri, result)
  146.  
  147.     def _save_dir_cb (self, handle, files, exception, (db, entry, pixbuf)):
  148.         artist, album = [db.entry_get (entry, x) for x in [rhythmdb.PROP_ARTIST, rhythmdb.PROP_ALBUM]]
  149.         for f in files:
  150.             if f.mime_type.split ("/")[0] in ["image", "x-directory"]:
  151.                 continue
  152.             uri = str (self.uri.parent.append_file_name (f.name))
  153.             u_entry = db.entry_lookup_by_location (uri)
  154.             if u_entry:
  155.                 u_artist, u_album = [db.entry_get (u_entry, x) for x in [rhythmdb.PROP_ARTIST, rhythmdb.PROP_ALBUM]]
  156.                 if album != u_album:
  157.                     print "Not saving local art; encountered media with different album (%s, %s, %s)" % (uri, u_artist, u_album)
  158.                     handle.cancel ()
  159.                     return
  160.                 continue
  161.             print "Not saving local art; encountered unknown file (%s)" % uri
  162.             handle.cancel ()
  163.             return
  164.         if exception:
  165.             if issubclass (exception, gnomevfs.EOFError):
  166.                 art_uri = str (self.uri.parent.append_file_name (ART_SAVE_NAME))
  167.                 print "Saving local art to \"%s\"" % art_uri
  168.                 rb.Coroutine (self.pixbuf_save, pixbuf, art_uri).begin ()
  169.             else:
  170.                 print "Error reading \"%s\": %s" % (self.uri.parent, exception)
  171.  
  172.     def save_pixbuf (self, db, entry, pixbuf):
  173.         self.uri = None
  174.         try:
  175.             self.uri = gnomevfs.URI (entry.get_playback_uri())
  176.         except TypeError:
  177.             pass
  178.  
  179.         if self.uri is None or self.uri.scheme == 'http':
  180.             print 'not saving local art for %s' % self.uri
  181.             return
  182.  
  183.         print 'checking whether to save local art for %s' % self.uri
  184.         gnomevfs.async.load_directory (self.uri.parent, self._save_dir_cb, LOAD_DIRECTORY_FLAGS, ITEMS_PER_NOTIFICATION, data=(db, entry, pixbuf))
  185.